import torch


A = torch.randn(5,5,5)
print("-"*20)
# 创建 range(27) 的数字
numbers = list(range(125))
# 将数字转换为 PyTorch 张量
tensor = torch.tensor(numbers)
# 调整形状为 [3, 3, 3]
tensor = tensor.view(5, 5, 5)
# print(tensor)
# print("-"*20)
# print(tensor[:-1])
# print(tensor[1:])
print("-"*20)
list_res = []
for a, b in zip(tensor[:-1], tensor[1:]):
    list_res.append(a)
    list_res.append(b)
    print(a)
    print(b)
    print("-"*20)
print(list_res, len(list_res))

units = [512, 512, 512]
for i, j in zip(units[:-1], units[1:]):
    print(i, j)

